package com.w3c.pragmatic.domain.res;

import com.github.pagehelper.PageInfo;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.experimental.Accessors;

import java.io.Serializable;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * ...
 *
 * @author wangdong
 * @since 2021/12/10 13:59
 */
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
@Accessors(chain = true)
public class PageRes<T> implements Serializable {

  private static final long serialVersionUID = -8800028898346527257L;

  /** 当前页（默认1） */
  private int current;

  /** 页大小（默认10） */
  private int size;

  /** 总页数 */
  private int pages;

  /** 总记录数 */
  private long total;

  /** 结果集列表 */
  private List<T> records;

  /**
   * 数据转化将page包装成PageInfo
   *
   * @param page 分页对象
   * @param <T> 对象类型
   * @return PageInfo
   */
  public static <T> PageRes<T> toPageRes(PageInfo<T> page) {
    PageRes<T> pageRes = new PageRes<>();
    pageRes.setCurrent(page.getPageNum());
    pageRes.setSize(page.getSize());
    pageRes.setPages(page.getPages());
    pageRes.setTotal(page.getTotal());
    pageRes.setRecords(page.getList());
    return pageRes;
  }

  /**
   * 数据转化将page包装成PageInfo
   *
   * @param page 分页对象
   * @param <F> 对象类型,entity
   * @param <T> 目标类型,dto\vo
   * @return PageInfo
   */
  public static <F, T> PageRes<T> toPageRes(PageInfo<F> page, Function<F, T> trans) {
    PageRes<T> pageRes = new PageRes<>();
    pageRes.setCurrent(page.getPageNum());
    pageRes.setSize(page.getSize());
    pageRes.setPages(page.getPages());
    pageRes.setTotal(page.getTotal());
    List<T> targetRecords = page.getList().stream().map(trans).collect(Collectors.toList());
    pageRes.setRecords(targetRecords);
    return pageRes;
  }
}
